All files / web/src/app/api/flowcharts/[id]/embedding route.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import { eq } from 'drizzle-orm'
import { NextResponse } from 'next/server'
import { withAuth } from '@/lib/auth/withAuth'
import { db, schema } from '@/db'
import { generateFlowchartEmbeddings, EMBEDDING_VERSION } from '@/lib/flowcharts/embedding'
import { invalidateEmbeddingCache } from '@/lib/flowcharts/embedding-search'

/**
 * POST /api/flowcharts/[id]/embedding
 *
 * Generate and store embeddings for a single published flowchart.
 */
export const POST = withAuth(async (_request, { params }) => {
  try {
    const { id } = (await params) as { id: string }

    const fc = await db.query.teacherFlowcharts.findFirst({
      where: eq(schema.teacherFlowcharts.id, id),
      columns: {
        id: true,
        title: true,
        description: true,
        difficulty: true,
        status: true,
      },
    })

    if (!fc || fc.status !== 'published') {
      return NextResponse.json({ error: 'Flowchart not found' }, { status: 404 })
    }

    const { embedding, promptEmbedding } = await generateFlowchartEmbeddings({
      title: fc.title,
      description: fc.description,
      topicDescription: null,
      difficulty: fc.difficulty,
    })

    await db
      .update(schema.teacherFlowcharts)
      .set({
        embedding,
        promptEmbedding,
        embeddingVersion: EMBEDDING_VERSION,
      })
      .where(eq(schema.teacherFlowcharts.id, id))

    invalidateEmbeddingCache()

    return NextResponse.json({ success: true })
  } catch (error) {
    console.error('Failed to generate embedding:', error)
    return NextResponse.json(
      { error: 'Failed to generate embedding', details: String(error) },
      { status: 500 }
    )
  }
})